/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package lintulaskenta.tipuapiValidointi;

import java.io.IOException;
import java.net.Socket;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.UnknownHostException;
import java.security.KeyManagementException;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.security.NoSuchAlgorithmException;
import java.security.UnrecoverableKeyException;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;

import javax.net.ssl.SSLContext;
import javax.net.ssl.TrustManager;
import javax.net.ssl.X509TrustManager;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.HttpVersion;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.HttpClient;
import org.apache.http.client.ResponseHandler;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.conn.scheme.PlainSocketFactory;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.scheme.SchemeRegistry;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.conn.tsccm.ThreadSafeClientConnManager;
import org.apache.http.params.BasicHttpParams;
import org.apache.http.params.HttpParams;
import org.apache.http.params.HttpProtocolParams;
import org.apache.http.protocol.HTTP;
import org.apache.http.util.EntityUtils;

/**
 *
 * @author jukkapajunen
 * Kopioitu pitkälti Esko Piiraisen
 * Luomus-filuista.
 */
public class HttpClientService {

	private final HttpClient httpclient;

	/**
	 * Returns a new HttpClient for site that requires HTTPBasic Authentication
	 * @param address
	 * @param username
	 * @param password
	 * @throws URISyntaxException
	 */
	public HttpClientService(String address, String username, String password) throws URISyntaxException {
		URI uri = new URI(address);
		DefaultHttpClient httpclient = (DefaultHttpClient) HttpClientUtil.getNewHttpClient();
		httpclient.getCredentialsProvider().setCredentials(
				new AuthScope(uri.getHost(), uri.getPort()),
				new UsernamePasswordCredentials(username, password));

		this.httpclient = httpclient;
	}

	/**
	 * Returns a new DefaultHttpClient
	 */
	public HttpClientService() {
		this.httpclient = HttpClientUtil.getNewHttpClient();
	}

	public void close() {
		httpclient.getConnectionManager().shutdown();
	}

	/**
	 * Returns response to HttpGet request as a XML Document.
	 * @see Document
	 * @param request
	 * @return
	 * @throws IOException
	 * @throws ClientProtocolException
	 */
	public Document contentAsDocument(HttpGet request) throws IOException, ClientProtocolException {
		String response = contentAsString(request);
		XMLReader reader = new XMLReader();
		Document doc = reader.parse(response);
		return doc;
	}

	/**
	 * Returns reponse to HttpGet request as a String.
	 * @param request
	 * @return
	 * @throws IOException
	 * @throws ClientProtocolException
	 */
	public String contentAsString(HttpGet request) throws IOException, ClientProtocolException {
		ResponseHandler<String> handler = contentAsStringHandler();
		String response = httpclient.execute(request, handler);
		return response;
	}

	/**
	 * Returns response to HttpPost request as a String.
	 * @param request
	 * @return
	 * @throws IOException
	 * @throws ClientProtocolException
	 */
	public String contentAsString(HttpPost request) throws IOException, ClientProtocolException {
		ResponseHandler<String> handler = contentAsStringHandler();
		String response = httpclient.execute(request, handler);
		return response;
	}

	private ResponseHandler<String> contentAsStringHandler() {
		ResponseHandler<String> handler = new ResponseHandler<String>() {
			@Override
			public String handleResponse(HttpResponse response) throws ClientProtocolException, IOException {
				HttpEntity entity = response.getEntity();
				if (entity != null) {
					return EntityUtils.toString(entity);
				} else {
					return null;
				}
			}
		};
		return handler;
	}

	/**
	 * Executes a request and returns a HttpResponse.
	 * @ss {@link HttpResponse}
	 * @param request
	 * @return
	 * @throws ClientProtocolException
	 * @throws IOException
	 */
	public HttpResponse execute(HttpPost request) throws ClientProtocolException, IOException {
		return httpclient.execute(request);
	}

	public static class HttpClientUtil {

		@SuppressWarnings("deprecation")
		public static HttpClient getNewHttpClient() {
			try {
				KeyStore trustStore = KeyStore.getInstance(KeyStore.getDefaultType());
				trustStore.load(null, null);

				SSLSocketFactory sf = new MySSLSocketFactory(trustStore);
				sf.setHostnameVerifier(SSLSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER);

				HttpParams params = new BasicHttpParams();
				HttpProtocolParams.setVersion(params, HttpVersion.HTTP_1_1);
				HttpProtocolParams.setContentCharset(params, HTTP.UTF_8);

				SchemeRegistry registry = new SchemeRegistry();
				registry.register(new Scheme("http", PlainSocketFactory.getSocketFactory(), 80));
				registry.register(new Scheme("https", sf, 443));

				ClientConnectionManager ccm = new ThreadSafeClientConnManager(params, registry);

				return new DefaultHttpClient(ccm, params);
			} catch (Exception e) {
				return new DefaultHttpClient();
			}
		}

		private static class MySSLSocketFactory extends SSLSocketFactory {
			SSLContext sslContext = SSLContext.getInstance("TLS");

			public MySSLSocketFactory(KeyStore truststore) throws NoSuchAlgorithmException, KeyManagementException, KeyStoreException, UnrecoverableKeyException {
				super(truststore);

				TrustManager tm = new X509TrustManager() {
					@Override
					public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {
					}

					@Override
					public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {
					}

					@Override
					public X509Certificate[] getAcceptedIssuers() {
						return null;
					}
				};

				sslContext.init(null, new TrustManager[] { tm }, null);
			}

			@Override
			public Socket createSocket(Socket socket, String host, int port, boolean autoClose) throws IOException, UnknownHostException {
				return sslContext.getSocketFactory().createSocket(socket, host, port, autoClose);
			}

			@Override
			public Socket createSocket() throws IOException {
				return sslContext.getSocketFactory().createSocket();
			}
		}
	}

}
